package com.cloud.controller;

import cn.hutool.core.date.DateUtil;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.cloud.common.domain.Result;
import com.cloud.common.entity.PageEntity;
import com.cloud.common.entity.TableDataInfo;
import com.cloud.common.entity.TableSupport;
import com.cloud.utils.SQLCheckUtils;
import com.cloud.utils.StringUtils;
import org.apache.commons.lang3.ObjectUtils;
import org.springframework.web.bind.WebDataBinder;
import org.springframework.web.bind.annotation.InitBinder;

import java.beans.PropertyEditorSupport;
import java.util.Date;
import java.util.List;
import java.util.Map;

/**
 * web层通用数据处理
 *
 * @author yzj
 */
public class BaseController {

    /**
     * 将前台传递过来的日期格式的字符串，自动转化为Date类型
     */
    @InitBinder
    public void initBinder(WebDataBinder binder) {
        // Date 类型转换
        binder.registerCustomEditor(Date.class, new PropertyEditorSupport() {
            @Override
            public void setAsText(String time) {
                setValue(DateUtil.parse(time));
            }
        });
    }

    /**
     * 设置请求分页数据
     */
    protected void startPage() {
        PageEntity pageEntity = TableSupport.buildPageRequest();
        Integer pageNum = pageEntity.getPageNum();
        Integer pageSize = pageEntity.getPageSize();
        if (StringUtils.isNotNull(pageNum) && StringUtils.isNotNull(pageSize)) {
            String orderBy = SQLCheckUtils.escapeOrderBySql(pageEntity.getOrderBy());
            PageHelper.startPage(pageNum, pageSize, orderBy);
        }
    }

    /**
     * 设置请求分页数据,不进行Count查询
     */
    protected PageEntity startPageBanCount() {
        PageEntity pageEntity = TableSupport.buildPageRequest();
        if (ObjectUtils.isEmpty(pageEntity.getPageNum())) {
            pageEntity.setPageNum(1);
        }
        if (ObjectUtils.isEmpty(pageEntity.getPageSize())) {
            pageEntity.setPageSize(10);
        }
        PageHelper.startPage(pageEntity.getPageNum(), pageEntity.getPageSize(), false);
        return pageEntity;
    }

    /**
     * 响应请求分页数据
     */
    protected Result getDataTable(List<?> list) {
        TableDataInfo tableData = new TableDataInfo();
        PageInfo page = new PageInfo(list);
        tableData.setCurrent(page.getPageNum());
        tableData.setSize(page.getPageSize());
        tableData.setTotal(page.getTotal());
        tableData.setRows(list);
        return Result.success("查询成功", tableData);
    }

    /**
     * 响应请求分页数据
     */
    protected Result getDataTable(PageEntity pageEntity, int total, List<?> list) {
        TableDataInfo tableData = new TableDataInfo();
        tableData.setCurrent(pageEntity.getPageNum());
        tableData.setSize(pageEntity.getPageSize());
        tableData.setTotal(total);
        tableData.setRows(list);
        return Result.success("查询成功", tableData);
    }

    /**
     * 响应请求分页数据
     */
    protected Result getDataTableMap(List<?> list, Map<String, Object> resMap) {
        TableDataInfo tableData = new TableDataInfo();
        PageInfo page = new PageInfo(list);
        tableData.setCurrent(page.getPageNum());
        tableData.setSize(page.getPageSize());
        tableData.setTotal(page.getTotal());
        tableData.setRows(list);
        resMap.put("page", tableData);
        return Result.success("查询成功", resMap);
    }

    /**
     * 响应返回结果
     *
     * @param rows 影响行数
     * @return 操作结果
     */
    protected Result toResult(int rows) {
        if (rows > 0) {
            return Result.success("操作成功");
        } else {
            return Result.error("操作失败");
        }
    }

    /**
     * 响应返回结果
     *
     * @param isSuccess 影响行数
     * @return 操作结果
     */
    protected Result toResult(boolean isSuccess) {
        if (isSuccess) {
            return Result.success("操作成功");
        } else {
            return Result.error("操作失败");
        }
    }

}
